package top.guxx.springframework.core.io;

import cn.hutool.core.lang.Assert;
import top.guxx.springframework.utils.ClassUtils;

import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;

/**
 * @Author：Guxx
 * @Description：通过ClassLoader读取文件信息
 * @Date：2024/6/9 21:34
 * @Filename：ClassPathResource
 */
public class ClassPathResource implements Resource {

    private final String path;
    private ClassLoader classLoader;

    public ClassPathResource(String path) {
        this(path, (ClassLoader) null);
    }

    public ClassPathResource(String path, ClassLoader classLoader) {
        Assert.notNull(path, "Path must not be null");
        this.path = path;
        this.classLoader = (classLoader != null ? classLoader :  ClassUtils.getDefaultClassLoader());
    }

    @Override
    public InputStream getInputStream() throws IOException {
        InputStream is = classLoader.getResourceAsStream(path);
        if (is == null){
            throw new FileNotFoundException(
                    this.path + "cannot be opened because it does not exist!"
            );
        }
        return is;
    }
}
